# 导入模块
import os
import time
from selenium import webdriver
from selenium.webdriver.common.by import By


# 初始化数据
class Button:
    def __init__(self):
        self.driver = webdriver.Chrome()
        self.driver.maximize_window()
        url = 'file:///' + os.path.abspath('html') + '\\radio-checkbox.html'
        self.driver.get(url)

    # TODO 单选按钮
    def btu(self):
        # 定位按钮位置，点击
        self.driver.find_element(By.ID, 'boy').click()
        time.sleep(3)
        self.driver.find_element(By.ID, 'girl').click()

    # TODO 多选按钮 方式一
    def buts1(self):
        # 定位所有按钮位置
        # 定位页面中第二个From标签
        form = self.driver.find_elements(By.TAG_NAME, 'form')[1]
        # 获取到from中所有的input标签
        inputs = form.find_elements(By.TAG_NAME, 'input')

        # 遍历inputs
        for ipu in inputs:
            # ipu.click() 遍历出一个元素点击一下'
            time.sleep(1)
            ipu.click()

    # TODO 多选按钮 方式二
    def buts2(self):
        # 定位所有的按钮位置
        inputs = self.driver.find_elements(By.XPATH, '//input[@type="checkbox"]')

        # 遍历inputs
        for ipu in inputs:
            # ipu.click() 每遍出一个元素点击一下'
            time.sleep(1)
            ipu.click()

    def quit(self):
        self.driver.quit()


if __name__ == '__main__':
    pass
